/*
 * FurnitureTransferHandler.java 12 sept. 2006
 *
 * Sweet Home 3D, Copyright (c) 2006 Emmanuel PUYBARET / eTeks <info@eteks.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package com.eteks.sweethome3d.swing;

import java.awt.EventQueue;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import javax.swing.JComponent;

import com.eteks.sweethome3d.model.Home;
import com.eteks.sweethome3d.model.HomePieceOfFurniture;
import com.eteks.sweethome3d.model.Level;
import com.eteks.sweethome3d.model.Selectable;
import com.eteks.sweethome3d.viewcontroller.ContentManager;
import com.eteks.sweethome3d.viewcontroller.HomeController;

/**
 * Home furniture transfer handler.
 * @author Emmanuel Puybaret
 */
public class FurnitureTransferHandler extends LocatedTransferHandler
{
	private final Home home;
	private final ContentManager contentManager;
	private final HomeController homeController;
	private List<HomePieceOfFurniture> copiedFurniture;
	private String copiedCSV;
	
	/**
	 * Creates a handler able to transfer home furniture.
	 */
	public FurnitureTransferHandler(Home home, ContentManager contentManager, HomeController homeController)
	{
		this.home = home;
		this.contentManager = contentManager;
		this.homeController = homeController;
	}
	
	/**
	 * Returns <code>COPY_OR_MOVE</code>.
	 */
	@Override
	public int getSourceActions(JComponent source)
	{
		return COPY_OR_MOVE;
	}
	
	/**
	 * Returns a {@link HomeTransferableList transferable object}
	 * that contains a copy of the selected furniture in home. 
	 */
	@Override
	protected Transferable createTransferable(JComponent source)
	{
		this.copiedFurniture = Home.getFurnitureSubList(this.home.getSelectedItems());
		final Transferable transferable = new HomeTransferableList(this.copiedFurniture);
		if (source instanceof FurnitureTable)
		{
			// Create a text that describes furniture in CSV format
			this.copiedCSV = ((FurnitureTable) source).getClipboardCSV();
			// Create a transferable that contains copied furniture and its CSV description 
			return new Transferable()
			{
				public Object getTransferData(DataFlavor flavor) throws UnsupportedFlavorException, IOException
				{
					if (DataFlavor.stringFlavor.equals(flavor))
					{
						return copiedCSV;
					}
					else
					{
						return transferable.getTransferData(flavor);
					}
				}
				
				public DataFlavor[] getTransferDataFlavors()
				{
					ArrayList<DataFlavor> dataFlavors = new ArrayList<DataFlavor>(
							Arrays.asList(transferable.getTransferDataFlavors()));
					dataFlavors.add(DataFlavor.stringFlavor);
					return dataFlavors.toArray(new DataFlavor[dataFlavors.size()]);
				}
				
				public boolean isDataFlavorSupported(DataFlavor flavor)
				{
					return transferable.isDataFlavorSupported(flavor) || DataFlavor.stringFlavor.equals(flavor);
				}
			};
		}
		else
		{
			return transferable;
		}
	}
	
	/**
	 * Removes the copied element once moved.
	 */
	@Override
	protected void exportDone(JComponent source, Transferable data, int action)
	{
		if (action == MOVE)
		{
			this.homeController.cut(copiedFurniture);
		}
		this.copiedFurniture = null;
		this.copiedCSV = null;
		this.homeController.enablePasteAction();
	}
	
	/**
	 * Returns <code>true</code> if flavors contains 
	 * {@link HomeTransferableList#HOME_FLAVOR HOME_FLAVOR} flavor
	 * or <code>DataFlavor.javaFileListFlavor</code> flavor.
	 */
	@Override
	public boolean canImportFlavor(DataFlavor[] flavors)
	{
		Level selectedLevel = this.home.getSelectedLevel();
		List<DataFlavor> flavorList = Arrays.asList(flavors);
		return (selectedLevel == null || selectedLevel.isViewable())
				&& (flavorList.contains(HomeTransferableList.HOME_FLAVOR)
						|| flavorList.contains(DataFlavor.javaFileListFlavor));
	}
	
	/**
	 * Add to home the furniture contained in <code>transferable</code>.
	 */
	@Override
	public boolean importData(JComponent destination, Transferable transferable)
	{
		if (canImportFlavor(transferable.getTransferDataFlavors()))
		{
			try
			{
				List<DataFlavor> flavorList = Arrays.asList(transferable.getTransferDataFlavors());
				if (flavorList.contains(HomeTransferableList.HOME_FLAVOR))
				{
					List<Selectable> items = (List<Selectable>) transferable
							.getTransferData(HomeTransferableList.HOME_FLAVOR);
					List<HomePieceOfFurniture> furniture = Home.getFurnitureSubList(items);
					if (isDrop())
					{
						this.homeController.drop(furniture, 0, 0);
					}
					else
					{
						this.homeController.paste(furniture);
					}
					return true;
				}
				else
				{
					List<File> files = (List<File>) transferable.getTransferData(DataFlavor.javaFileListFlavor);
					final List<String> importableModels = getModelContents(files, this.contentManager);
					EventQueue.invokeLater(new Runnable()
					{
						public void run()
						{
							homeController.dropFiles(importableModels, 0, 0);
						}
					});
					return !importableModels.isEmpty();
				}
			}
			catch (UnsupportedFlavorException ex)
			{
				throw new RuntimeException("Can't import", ex);
			}
			catch (IOException ex)
			{
				throw new RuntimeException("Can't access to data", ex);
			}
		}
		else
		{
			return false;
		}
	}
}
